| Name | Version | Summary | date |
| jaxsim |
0.6.1 |
A differentiable physics engine and multibody dynamics library for control and robot learning. |
2025-01-20 09:19:08 |
| sbiax |
0.0.55 |
Fast, parallel and lightweight simulation-based inference in JAX. |
2025-01-17 14:43:07 |
| invrs-opt |
0.10.4 |
Algorithms for inverse design |
2025-01-15 21:04:28 |
| invrs-gym |
1.4.6 |
A collection of inverse design challenges |
2025-01-15 21:03:42 |
| cryojax |
0.3.3 |
Cryo-EM image simulation and analysis powered by JAX |
2025-01-15 14:57:43 |
| fmmax |
0.13.0 |
Fourier modal method with Jax |
2025-01-15 05:29:49 |
| agjax |
0.3.5 |
A jax wrapper for autograd-differentiable functions. |
2025-01-15 05:01:27 |
| reinforced-lib |
1.2.0 |
Reinforcement learning library |
2025-01-13 17:10:05 |
| kozax |
0.0.2 |
Genetic programming library in JAX. |
2025-01-10 13:11:07 |
| abmax |
0.0.1 |
ABM package in JAX. |
2025-01-10 11:35:01 |
| Kozax |
0.0.1 |
Genetic programming library in JAX. |
2025-01-10 11:11:25 |
| sbgm |
0.0.21 |
Score-based Diffusion models in JAX. |
2025-01-09 17:59:54 |
| adam-robotics |
0.3.3 |
Automatic Differentiation for rigid-body-dynamics AlgorithMs |
2025-01-08 17:28:15 |
| pastax |
0.0.2 |
**P**arameterizable **A**uto-differentiable **S**imulators of ocean **T**rajectories in j**AX**. |
2025-01-06 11:07:02 |
| reax |
0.3.0 |
REAX: A simple training framework for JAX-based projects |
2025-01-04 23:17:35 |
| optimistix |
0.0.10 |
Nonlinear optimisation in JAX and Equinox. |
2025-01-03 23:48:28 |
| triton-model-navigator |
0.13.1 |
Triton Model Navigator: An inference toolkit for optimizing and deploying machine learning models and pipelines on the Triton Inference Server and PyTriton. |
2025-01-03 12:24:17 |
| sympy2jax |
0.0.6 |
Turn SymPy expressions into trainable JAX expressions. |
2025-01-02 00:46:32 |
| psgd-jax |
0.2.9 |
An implementation of PSGD optimizer in JAX. |
2024-12-31 22:15:55 |
| diffrax |
0.6.2 |
GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX. |
2024-12-24 16:50:08 |